import os

from dotenv import load_dotenv
from langchain_core.messages import SystemMessage, HumanMessage
from langchain_core.output_parsers import StrOutputParser
from langchain_openai import ChatOpenAI

# 加载.env文件
load_dotenv()

# 调用大语言模型
# 创建模型
model = ChatOpenAI(
	model='qwen-plus',
	api_key=os.getenv("OPENAI_API_KEY"),
	base_url=os.getenv("BASE_URL"),
)

# 2、准备prompt
msg = [
	SystemMessage(content='请将以下的内容翻译成英语'),
	HumanMessage(content='你好，请问你要去哪里？')
	# HumanMessage(content='java可以干些什么?')
]

# result = model.invoke(msg).content
# print(result)

# 获取解析器
parser = StrOutputParser()

# 获得链
chain = model | parser

print(chain.invoke(msg))
